Skip to content

[DO NOT MERGE] [FEA] KMeans - Batched (Streaming) Predict for Host Matrices#1962

Draft
tarang-jain wants to merge 2 commits intorapidsai:mainfrom
tarang-jain:batched-predict
Draft

[DO NOT MERGE] [FEA] KMeans - Batched (Streaming) Predict for Host Matrices#1962
tarang-jain wants to merge 2 commits intorapidsai:mainfrom
tarang-jain:batched-predict

Conversation

@tarang-jain
Copy link
Copy Markdown
Contributor

The predict function was separated out from #1886 into this PR. This PR is exploratory and we must evaluate the need for a host side predict function because the streaming of batches to device and calling of device side predict can be done by the user with the device side predict API itself.

@copy-pr-bot
Copy link
Copy Markdown

copy-pr-bot bot commented Mar 28, 2026

Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually.

Contributors can view more details about this message here.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

Status: No status

Development

Successfully merging this pull request may close these issues.

1 participant